from langchain_openai import ChatOpenAI
from globle import envs
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate

system_prompt = '''
你是一个判断两句话是否是相似句子的助手,只需要回答相似或不相似
'''

sentences = [('股票市场今日大涨，投资者乐观。', '持续上涨的市场让投资者感到满意。'),
('油价大幅下跌，能源公司面临挑战。', '未来智能城市的建设趋势愈发明显。'),
('利率上升，影响房地产市场。', '高利率对房地产有一定冲击。')]

prompt_template = ChatPromptTemplate.from_messages(
    [("system", system_prompt)
    ,("user", "句子1：{text1}，句子2：{text2}")])

# 创建模型
model = ChatOpenAI(
    model="deepseek-chat",  # 或具体模型名
    base_url="https://api.deepseek.com/v1",  # DeepSeek API 地址

)
# 创建解析器
parser = StrOutputParser()
# 创建链
chain = prompt_template | model | parser
# 调用链
for sent in sentences:
    r = chain.invoke({"text1":sent[0],"text2":sent[1]})
    print(r)
